from langchain_core.messages import SystemMessage, HumanMessage
from config.load_key import load_key
from langchain.chat_models import init_chat_model
import os

# 判断 OPENAI_API_KEY 是否存在，不存在则从 config/Keys.json 中读取
if not os.environ.get("OPENAI_API_KEY"):
    os.environ["OPENAI_API_KEY"] = load_key("siliconflow_api_key")
# 初始化聊天模型
model = init_chat_model("Qwen/Qwen3-8B", model_provider="openai", base_url=load_key("siliconflow_base_url"))
# 输入消息
messages = [
    SystemMessage("帮我将英文翻译成中文"),
    HumanMessage("Hello, how are you?"),
]
# 输出翻译结果
print(model.invoke(messages).content)
